
#pragma once
#include <vector>
#include <pthread.h>
#include <iostream>
#include <semaphore.h>

using namespace std;
namespace ns_ring_queue
{
    template <class T>
    class RingQueue
    {
    private:
        vector<T> _ring_queue; // 环形队列
        int _cap;              // 容量
        sem_t _black_sem;      // 生产者关系空位置资源
        sem_t _data_sem;       // 消费者关心数据资源
        int _c_step;           // 消费者走到哪
        int _p_step;
        pthread_mutex_t _c_mtx;
        pthread_mutex_t _p_mtx;

    public:
        RingQueue(int cap = 3)
            : _ring_queue(cap), _cap(cap)
        {
            sem_init(&_black_sem, 0, cap);
            sem_init(&_data_sem, 0, 0);
            _c_step = _p_step = 0; // 从0开始走

            pthread_mutex_init(&_c_mtx, nullptr);
            pthread_mutex_init(&_p_mtx, nullptr);
        }
        ~RingQueue()
        {
            sem_destroy(&_black_sem);
            sem_destroy(&_data_sem);
            pthread_mutex_destroy(&_c_mtx);
            pthread_mutex_destroy(&_p_mtx);

        }
    public:
        void Push(const T& in)
        {
            //生产接口
            sem_wait(&_black_sem); //并行的先分配好信号量

            pthread_mutex_lock(&_p_mtx);//再分配锁

            _ring_queue[_p_step]=in;//生产到p_step位置上
            _p_step++;
            _p_step%=_cap;

            pthread_mutex_unlock(&_p_mtx);

            sem_post(&_data_sem);
        }

         void Pop(T* out)
        {
            //消费接口
            sem_wait(&_data_sem);
            pthread_mutex_lock(&_c_mtx);

            *out=_ring_queue[_c_step]; //拿c_step上的数据
            _c_step++;
            _c_step%=_cap;

            pthread_mutex_unlock(&_c_mtx);

            sem_post(&_black_sem);
        }

    };
}